import com.jfinal.kit.PathKit;
import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;
import org.junit.Test;

import javax.sql.DataSource;

/**
 * @authc yaowk
 * 2017/4/21
 */
public class GeneratorMain {

    public static DataSource getDataSource() {
        Prop prop = PropKit.use("jdbc.properties");
        DruidPlugin druidPlugin = new DruidPlugin(prop.get("jdbcUrl"), prop.get("user"), prop.get("password"), prop.get("driverClass"));
        druidPlugin.start();
        return druidPlugin.getDataSource();
    }

    public static void main(String[] args) {
        // base model 所使用包名
        String baseModelPackageName = "com.yaowk.email.model.base";
        // base model 文件保存路径
        String baseModelOutputDir = PathKit.getWebRootPath() + "/src/main/java/com/yaowk/email/model/base";
        // model 所使用的包名（MappingKit 默认使用包名）
        String modelPackageName = "com.yaowk.email.model";
        // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
        String modelOutputDir = baseModelOutputDir + "/..";
        // 创建生成器
        DataSource dataSource = getDataSource();
        Generator gernerator = new Generator(dataSource, baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);
        gernerator.setDialect(new MysqlDialect());
        gernerator.setMetaBuilder(new EmailMetaBuilder(dataSource));
        gernerator.setGenerateChainSetter(true);
        // 设置是否在 Model 中生成 dao 对象
        gernerator.setGenerateDaoInModel(true);
        // 设置是否生成字典文件
        gernerator.setGenerateDataDictionary(false);
        // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
//		gernerator.setRemovedTableNamePrefixes("t_","sys_");
        // 生成
        gernerator.generate();
    }

    @Test
    public void pathTest() {
        System.out.println(PathKit.getWebRootPath());
        System.out.println(PathKit.getRootClassPath());
        System.out.println(PathKit.getPackagePath(this));
    }
}
